package com.sky.service.impl;

import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.sky.constant.MessageConstant;
import com.sky.constant.StatusConstant;
import com.sky.dto.CategoryDTO;
import com.sky.dto.SetmealDTO;
import com.sky.dto.SetmealPageQueryDTO;
import com.sky.entity.*;
import com.sky.exception.DeletionNotAllowedException;
import com.sky.exception.SetmealEnableFailedException;
import com.sky.mapper.DishFlavorMapper;
import com.sky.mapper.DishMapper;
import com.sky.mapper.SetmealDishMapper;
import com.sky.mapper.SetmealMapper;
import com.sky.result.PageResult;
import com.sky.result.Result;
import com.sky.service.CategoryService;
import com.sky.service.DishService;
import com.sky.service.SetmealService;
import com.sky.vo.DishItemVO;
import com.sky.vo.DishVO;
import com.sky.vo.SetmealDishVo;
import com.sky.vo.SetmealVO;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.RequestParam;

import java.util.ArrayList;
import java.util.List;

@Service
public class SetmealServiceImpl implements SetmealService {

    @Autowired
    private SetmealMapper setmealMapper;

    @Autowired
    private SetmealDishMapper setmealDishMapper;

    @Autowired
    private DishService dishService;

    @Autowired
    private CategoryService categoryService;

    @Autowired
    private DishMapper dishMapper;

    /**
     * 新增套餐，同时需要保存套餐和菜品的关联关系
     * @param setmealDTO
     */
    @Transactional
    public void addSetmealWithDish(SetmealDTO setmealDTO) {
        Setmeal setmeal = new Setmeal();
        BeanUtils.copyProperties(setmealDTO, setmeal);

        //向套餐表插入数据
        setmealMapper.insert(setmeal);

        //获取生成的套餐id
        Long setmealId = setmeal.getId();

        List<SetmealDish> setmealDishes = setmealDTO.getSetmealDishes();
        for (SetmealDish setmealDish : setmealDishes) {
            setmealDish.setSetmealId(setmealId);
        }
        //保存套餐和菜品的关联关系
        setmealDishMapper.insertBatch(setmealDishes);
    }

    /**
     * 套餐分页查询
     * @param setmealPageQueryDTO
     */
    @Override
    public PageResult setMealPageQuery(SetmealPageQueryDTO setmealPageQueryDTO) {
        PageHelper.startPage(setmealPageQueryDTO.getPage(), setmealPageQueryDTO.getPageSize());
        Page<SetmealVO> page = setmealMapper.setMealPageQuery(setmealPageQueryDTO);
        for (SetmealVO setmealVO : page.getResult()) {
            List<SetmealDishVo> setmealDishVoes = new ArrayList<>();
            // 获取套餐id
            Long id = setmealVO.getId();
            List<Long> ids = new ArrayList<>();
            ids.add(id);
            // 根据套餐id获取套餐菜品关系数据
            List<SetmealDish> setmealDishes = setmealDishMapper.getSetmealIdsBySetmealIds(ids);

            for (SetmealDish setmealDish : setmealDishes) {
                SetmealDishVo setmealDishVo = new SetmealDishVo();
                BeanUtils.copyProperties(setmealDish, setmealDishVo);
                // 根据菜品id查询菜品
                setmealDishVo.setDish(dishService.getDishWithFlavorById(setmealDish.getDishId()));
                setmealDishVoes.add(setmealDishVo);
            }
            setmealVO.setSetmealDishes(setmealDishVoes);

        }
        return new PageResult(page.getTotal(), page.getResult());
    }

    /**
     * 批量删除套餐
     * @param ids
     */
    @Transactional
    public void deleteSetmeal(List<Long> ids) {
        ids.forEach((id) -> {
            Setmeal setmeal = setmealMapper.getById(id);
            if(setmeal.getStatus() == StatusConstant.ENABLE) {
                // 起售中的套餐不能删除
                throw new DeletionNotAllowedException(MessageConstant.SETMEAL_ON_SALE);
            }else {
                //删除套餐表中的数据
                setmealMapper.deleteById(id);
                // 删除套餐关系表中的数据
                setmealDishMapper.deleteBySetmealId(id);
            }
        });

    }

    /**
     * 根据id查询套餐
     * @param id
     */
    public SetmealVO getSetmealById(Long id) {
        Setmeal setmeal = setmealMapper.getById(id);
        SetmealVO setmealVO = new SetmealVO();
        BeanUtils.copyProperties(setmeal, setmealVO);

        // 根据分类id查询分类名称
        Long categoryId = setmeal.getCategoryId();
        if(categoryId != null) {
            Category category = categoryService.getById(categoryId);
            if(category != null) {
                setmealVO.setCategoryName(category.getName());
            }
        }

        List<SetmealDishVo> setmealDishVoes = new ArrayList<>();
        // 根据套餐id获取套餐菜品关系数据
        List<Long> ids = new ArrayList<>();
        ids.add(id);
        List<SetmealDish> setmealDishes = setmealDishMapper.getSetmealIdsBySetmealIds(ids);

        for (SetmealDish setmealDish : setmealDishes) {
            SetmealDishVo setmealDishVo = new SetmealDishVo();
            BeanUtils.copyProperties(setmealDish, setmealDishVo);
            // 根据菜品id查询菜品
            DishVO dishVO = dishService.getDishWithFlavorById(setmealDish.getDishId());
            setmealDishVo.setDish(dishVO);
            setmealDishVoes.add(setmealDishVo);
        }
        setmealVO.setSetmealDishes(setmealDishVoes);
        return setmealVO;
    }

    /**
     * 修改套餐
     * @param setmealDTO
     */
    public void updateSetmeal(SetmealDTO setmealDTO) {
        Setmeal setmeal = new Setmeal();
        BeanUtils.copyProperties(setmealDTO, setmeal);

        // 修改套餐表，执行update
        setmealMapper.update(setmeal);

        List<SetmealDish> setmealDishes = setmealDTO.getSetmealDishes();
        // 获取套餐id
        Long setmealId = setmealDTO.getId();
        // 删除原有的套餐菜品关系数据
        setmealDishMapper.deleteBySetmealId(setmealId);
        setmealDishes.forEach(setmealDish -> {
            setmealDish.setSetmealId(setmealId);
        });
        //3、重新插入套餐和菜品的关联关系，操作setmeal_dish表，执行insert
        setmealDishMapper.insertBatch(setmealDishes);

    }

    /**
     * 套餐起售、停售
     * @param status
     * @param id
     */
    @Override
    public void startOrStop(Integer status, Long id) {
        //起售套餐时，判断套餐内是否有停售菜品，有停售菜品提示"套餐内包含未启售菜品，无法启售"
        if(status == StatusConstant.ENABLE){
            //select a.* from dish a left join setmeal_dish b on a.id = b.dish_id where b.setmeal_id = ?
            List<Dish> dishList = dishMapper.getBySetmealId(id);
            if(dishList != null && dishList.size() > 0){
                dishList.forEach(dish -> {
                    if(StatusConstant.DISABLE == dish.getStatus()){
                        throw new SetmealEnableFailedException(MessageConstant.SETMEAL_ENABLE_FAILED);
                    }
                });
            }
        }

        Setmeal setmeal = Setmeal.builder()
                .id(id)
                .status(status)
                .build();
        setmealMapper.update(setmeal);
    }

    /**
     * 条件查询
     * @param setmeal
     * @return
     */
    public List<Setmeal> list(Setmeal setmeal) {
        List<Setmeal> list = setmealMapper.list(setmeal);
        return list;
    }

    /**
     * 根据id查询菜品选项
     * @param id
     * @return
     */
    public List<DishItemVO> getDishItemById(Long id) {
        return setmealMapper.getDishItemBySetmealId(id);
    }
}
